//
// Created by fss on 22-12-14.
//
#include "data/tensor.hpp"
#include <gtest/gtest.h>
#include <armadillo>
#include <glog/logging.h>
#include "data/tensor.hpp"

TEST(test_tensor, create) {
  using namespace kuiper_infer;
  Tensor<float> tensor(3, 32, 32);
  ASSERT_EQ(tensor.channels(), 3);
  ASSERT_EQ(tensor.rows(), 32);
  ASSERT_EQ(tensor.cols(), 32);
  ASSERT_EQ(tensor.empty(), false);
}

//TEST(test_tensor, fill) {
//  using namespace kuiper_infer;
//  Tensor<float> tensor(3, 3, 3);
//  ASSERT_EQ(tensor.channels(), 3);
//  ASSERT_EQ(tensor.rows(), 3);
//  ASSERT_EQ(tensor.cols(), 3);
//
//  std::vector<float> values;
//  for (int i = 0; i < 27; ++i) {
//    values.push_back((float) i);
//  }
//  tensor.Fill(values);
//  LOG(INFO) << tensor.data();
//
//  int index = 0;
//  for (int c = 0; c < tensor.channels(); ++c) {
//    for (int c_ = 0; c_ < tensor.cols(); ++c_) {
//      for (int r = 0; r < tensor.rows(); ++r) {
//        ASSERT_EQ(values.at(index), tensor.at(c, c_, r));
//        index += 1;
//      }
//    }
//  }
//  LOG(INFO) << "Test1 passed!";
//}
//
//TEST(test_tensor, padding1) {
//  using namespace kuiper_infer;
//  Tensor<float> tensor(3, 3, 3);
//  ASSERT_EQ(tensor.channels(), 3);
//  ASSERT_EQ(tensor.rows(), 3);
//  ASSERT_EQ(tensor.cols(), 3);
//
//  tensor.Fill(1.f); // 填充为1
//  tensor.Padding({1, 1, 1, 1}, 0); // 边缘填充为0
//  ASSERT_EQ(tensor.rows(), 5);
//  ASSERT_EQ(tensor.cols(), 5);
//
//  int index = 0;
//  // 检查一下边缘被填充的行、列是否都是0
//  for (int c = 0; c < tensor.channels(); ++c) {
//    for (int c_ = 0; c_ < tensor.cols(); ++c_) {
//      for (int r = 0; r < tensor.rows(); ++r) {
//        if (c_ == 0 || r == 0) {
//          ASSERT_EQ(tensor.at(c, c_, r), 0);
//        }
//        index += 1;
//      }
//    }
//  }
//  LOG(INFO) << "Test2 passed!";
//}